package com.yc.cloud.openai.portal.enums;


import lombok.Getter;

import java.util.Objects;

@Getter
public enum ImageStyle {
    //写实人物风格
//    REALISTIC_PORTRAIT("REALISTIC_PORTRAIT", "majicMIX realistic 麦橘写实_v7"),
//    //自然风格
//    NATURAL_HARMONY("NATURAL_HARMONY", "极致写实风景摄影MAX_V1.0.safetensors"),
    //宫崎骏风格
    HAYAO_MIYAZAKI("HAYAO_MIYAZAKI", "SD1.5_宫崎骏动漫画风_V1.0.safetensors"),

    //3D可爱风格
    CUTE_3D("CUTE_3D", "SD1.5_3D可爱IP形象_v1.0.safetensors"),

    //赛博朋克
    CYBERPUNK("CYBERPUNK", "SD1.5_赛博朋克场景_v1.0.safetensors"),

    //胶片风
    FILM("FILM", "SD1.5_ FilmGirl胶片风.safetensors"),

    //胶片风
    ASIAN_GIRL("ASIAN_GIRL", "SD1.5_亚洲美女脸型_v1.safetensors"),

    //目前底模就是写实的,所以不需要用lora风格了
    REALISTIC("REALISTIC", "");

    private final String key;
    private final String value;

    ImageStyle(String key, String value) {
        this.key = key;
        this.value = value;
    }


    public static ImageStyle fromKey(String key) {
        for (ImageStyle provider : ImageStyle.values()) {
            if (Objects.equals(provider.getKey(), key)) {
                return provider;
            }
        }
        throw new IllegalArgumentException("Unknown provider key: " + key);
    }
}

